Color the correct shapes